Augmented reality and virtual reality rehabilitation